# 基础模型配置
MODEL_CONFIG = {
    "base_model": {
        "path": "sdxl/hnwsdxl.safetensors",
        "use_safetensors": True,
        "local_files_only": True,
        "enable_pag": True,
        "pag_applied_layers": ["mid"]
    },
    "controlnet": {
        "path": "controlnet-canny-sdxl-xinsir/",
        "local_files_only": True,
        "use_safetensors": True,
        "enable_pag": True,
        "pag_applied_layers": "mid"
    }
}

# LoRA模型配置
LORA_CONFIG = {
    "default": {
        "path": "lora_weight/sdxl-lora-linedraw.safetensors",
        "weight_name": "sdxl-lora-linedraw.safetensors",
        "adapter_name": "sdxl-lora-linedraw"
    },
    "ChineseStyle": {
        "path": "lora_weight/国风水彩模型v1_v1.safetensors",
        "weight_name": "国风水彩模型v1_v1.safetensors",
        "adapter_name": "国风水彩模型v1_v1"
    },
    "WarterColorPink": {
        "path": "lora_weight/watercolorpink3.safetensors",
        "weight_name": "watercolorpink3.safetensors",
        "adapter_name": "watercolorpink3"
    },
    "Lineart": {
        "path": "lora_weight/lineart.safetensors",
        "weight_name": "lineart.safetensors",
        "adapter_name": "lineart"
    },
    "ChildrenDraw1": {
        "path": "lora_weight/ChildrenDraw.safetensors",
        "weight_name": "ChildrenDraw.safetensors",
        "adapter_name": "ChildrenDraw"
    },
    "ChildrenDraw2": {
        "path": "lora_weight/ChildrenDraw.safetensors",
        "weight_name": "ChildrenDraw.safetensors",
        "adapter_name": "ChildrenDraw"
    },
    "ChildrenDraw3": {
        "path": "lora_weight/ChildrenDraw.safetensors",
        "weight_name": "ChildrenDraw.safetensors",
        "adapter_name": "ChildrenDraw"
    },
    "ChildrenDraw4": {
        "path": "lora_weight/ChildrenDraw.safetensors",
        "weight_name": "ChildrenDraw.safetensors",
        "adapter_name": "ChildrenDraw"
    },
    "ChildrenDraw5": {
        "path": "lora_weight/ChildrenDraw.safetensors",
        "weight_name": "ChildrenDraw.safetensors",
        "adapter_name": "ChildrenDraw"
    },
    "ChildrenDraw6": {
        "path": "lora_weight/ChildrenDraw6.safetensors",
        "weight_name": "ChildrenDraw6.safetensors",
        "adapter_name": "ChildrenDraw6"
    },
    "ChildrenDraw7": {
        "path": "lora_weight/ChildrenDraw7.safetensors",
        "weight_name": "ChildrenDraw7.safetensors",
        "adapter_name": "ChildrenDraw7"
    },
    "ChildrenLineDraw": {
        "path": "lora_weight/ChildrenLineDraw.safetensors",
        "weight_name": "ChildrenLineDraw.safetensors",
        "adapter_name": "ChildrenLineDraw"
    },
    "Anime": {
        "path": "lora_weight/anime.safetensors",
        "weight_name": "anime.safetensors",
        "adapter_name": "anime"
    }
}

# 风格相关参数配置
STYLE_CONFIG = {
    "default": {
        "prompt": "line draw, sketch",
        "guidance_scale": 3,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChineseStyle": {
        "prompt": "guo_feng",
        "guidance_scale": 2,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "WarterColorPink": {
        "prompt": "watercolor pink",
        "guidance_scale": 3,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "Lineart": {
        "prompt": "sketch",
        "guidance_scale": 3,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "Pixar": {
        "prompt": "pixar-style",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChildrenDraw1": {
        "prompt": "children-drawing,colorful drawing,whimsical drawing,vibrant drawing,cartoon,",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChildrenDraw2": {
        "prompt": "children-drawing,colorful drawing",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChildrenDraw3": {
        "prompt": "children-drawing,whimsical drawing",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChildrenDraw4": {
        "prompt": "children-drawing,vibrant drawing,",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChildrenDraw5": {
        "prompt": "children-drawing,cartoon,",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "ChildrenDraw6": {
        "prompt": "children's simple drawings，sketched style,",
        "guidance_scale": 3.5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.6
    },
    "ChildrenDraw7": {  
        "prompt": "Children's painting,",
        "guidance_scale": 2,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.8
    },
    "ChildrenLineDraw": {
        "prompt": "line drawing,colorful drawing,cartoon,",
        "guidance_scale": 5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    },
    "Anime": {
        "prompt": "Anime,",
        "guidance_scale": 3.5,
        "base_strength": 0.8,
        "controlnet_conditioning_scale": 0.7
    }
} 